MyPy error: Expected type in class pattern; found Any